package com.demo.stream_;

import org.junit.Test;

import java.util.ArrayList;
import java.util.Arrays;
import java.util.List;
import java.util.stream.Collectors;
import java.util.stream.Stream;

/**
 * @author gy
 * @date 2022/5/10
 * 映射，可以将一个流的 所有元素 按照一定的映射规则 映射到另一个流中
 * map：接收一个函数作为参数，该函数会被应用到每个元素上，并将其映射成一个 新的元素
 * flatMap：接收一个函数作为参数，将流的每个值换成另一个流，然后把所有流连接成一个流
 *
 * map 和 filter 的区别；
 * 映射 相当于 按照一定规则 修改流中每个元素
 * 筛选 相当于 按照一定规则 筛选出符合条件的元素
 */
public class StreamMap {
    /**
     * 英文字符串数组的元素全部改为大写。整数数组每个元素+3。
     */
    @Test
    public void m1(){
        // 英文字符串数组的元素全部改为大写。
        String[] strArr = { "abcd", "bcdd", "defde", "fTr" };
        List<String> strList = Arrays.stream(strArr).map(String::toUpperCase).collect(Collectors.toList());

        // 整数数组每个元素+3。
        List<Integer> intList = Arrays.asList(1, 3, 5, 7, 9, 11);
        List<Integer> intListNew = intList.stream().map(x -> x + 3).collect(Collectors.toList());

        System.out.println("每个元素大写：" + strList);
        System.out.println("每个元素+3：" + intListNew);

    }

    /**
     * 将员工的薪资全部增加1000
     * 根据业务需求来进行选择
     * 1、不改变原来员工集合的方式  重新创建了对象，用于操作自定义的对象，不会对原来的对象产生修改
     * 2、改变原来员工集合的方式 在原来集合的对象上面进行了修改
     */
    @Test
    public void m2(){
        List<Person> personList = new ArrayList<Person>();
        personList.add(new Person("Tom", 8900, 23, "male", "New York"));
        personList.add(new Person("Jack", 7000, 25, "male", "Washington"));
        personList.add(new Person("Lily", 7800, 21, "female", "Washington"));
        personList.add(new Person("Anni", 8200, 24, "female", "New York"));
        personList.add(new Person("Owen", 9500, 25, "male", "New York"));
        personList.add(new Person("Alisa", 7900, 26, "female", "New York"));

        // 不改变原来员工集合的方式
        List<Person> personListNew = personList.stream().map(person -> {
            // 可以对新的 对象进行自定义修改，不改变原来的集合
            Person personNew = new Person(person.getName(), person.getSalary(), 18, person.getSex(), null);
            personNew.setSalary(person.getSalary() + 10000);
            return personNew;
        }).collect(Collectors.toList());
        System.out.println("一次改动前：" + personList.get(0).getName() + "-->" + personList.get(0).getSalary());
        System.out.println("一次改动后：" + personListNew.get(0).getName() + "-->" + personListNew.get(0).getSalary());

        System.out.println("一次改动前完整数据：" + personList);
        System.out.println("一次改动后完整数据：" + personListNew);


        // 改变原来员工集合的方式（操作的是集合中的同一个对象）
        List<Person> personListNew2 = personList.stream().map(person -> {
            person.setSalary(person.getSalary() + 10000);
            return person;
        }).collect(Collectors.toList());
        System.out.println("二次改动前：" + personList.get(0).getName() + "-->" + personListNew.get(0).getSalary());
        System.out.println("二次改动后：" + personListNew2.get(0).getName() + "-->" + personListNew.get(0).getSalary());

        System.out.println("二次改动前完整数据：" + personList);
        System.out.println("二次改动后完整数据：" + personListNew2);

    }

    /**
     * 将两个字符数组合并成一个新的字符数组。
     * 接收一个函数作为参数，将流的每个值换成另一个流，然后把所有流连接成一个流
     */
    @Test
    public void m3(){
        List<String> list = Arrays.asList("m,k,l,a", "1,3,5,7");
        List<String> listNew = list.stream().flatMap(s -> {
            // 将每个元素转换成一个stream
            String[] split = s.split(",");
            Stream<String> s2 = Arrays.stream(split);
            return s2;
        }).collect(Collectors.toList());

        System.out.println("处理前的集合：" + list);
        System.out.println("处理后的集合：" + listNew);
    }
}
